package com.shm.leetcode;

/**
 * 211. 添加与搜索单词 - 数据结构设计
 * 请你设计一个数据结构，支持 添加新单词 和 查找字符串是否与任何先前添加的字符串匹配 。
 *
 * 实现词典类 WordDictionary ：
 *
 * WordDictionary() 初始化词典对象
 * void addWord(word) 将 word 添加到数据结构中，之后可以对它进行匹配
 * bool search(word) 如果数据结构中存在字符串与 word 匹配，则返回 true ；否则，返回  false 。word 中可能包含一些 '.' ，每个 . 都可以表示任何一个字母。
 *
 *
 * 示例：
 *
 * 输入：
 * ["WordDictionary","addWord","addWord","addWord","search","search","search","search"]
 * [[],["bad"],["dad"],["mad"],["pad"],["bad"],[".ad"],["b.."]]
 * 输出：
 * [null,null,null,null,false,true,true,true]
 *
 * 解释：
 * WordDictionary wordDictionary = new WordDictionary();
 * wordDictionary.addWord("bad");
 * wordDictionary.addWord("dad");
 * wordDictionary.addWord("mad");
 * wordDictionary.search("pad"); // return False
 * wordDictionary.search("bad"); // return True
 * wordDictionary.search(".ad"); // return True
 * wordDictionary.search("b.."); // return True
 *
 *
 * 提示：
 *
 * 1 <= word.length <= 500
 * addWord 中的 word 由小写英文字母组成
 * search 中的 word 由 '.' 或小写英文字母组成
 * 最调用多 50000 次 addWord 和 search
 */
class WordDictionary {
    boolean isEnd = false;
    WordDictionary next[] = new WordDictionary[26];

    /** Initialize your data structure here. */
    public WordDictionary() {

    }
    //插入新单词
    public void addWord(String word) {
        WordDictionary wd = this;
        char[] chars = word.toCharArray();
        for (char aChar : chars) {
            if (wd.next[aChar-'a']==null){
                wd.next[aChar-'a'] = new WordDictionary();
            }
           wd = wd.next[aChar-'a'];
        }
        wd.isEnd = true;
    }

    //模糊搜索
    public boolean search(String word) {
        WordDictionary wd  = this;
        return dfs(word,wd);
    }

    //因为.的存在，因此要进行dfs来达到模糊搜索的目的
    boolean dfs(String word,WordDictionary wd){
        char[] chars = word.toCharArray();
        for (int i = 0; i < chars.length; i++) {
            //如果说当前位置为. ，那么就对当前节点next[]中所有不为空的节点发起一次dfs
            //只要有一个满足匹配要求，那么就可以直接返回true
            if (chars[i]=='.'){
                for (int j = 0; j < 26; j++) {
                    if (wd.next[j]!=null&&dfs(word.substring(i+1),wd.next[j])){
                        return true;
                    }
                }
                return false;
            }
            if (wd.next[chars[i]-'a']==null){
                return false;
            }

            wd = wd.next[chars[i]-'a'];
        }
        return wd.isEnd;
    }
}

/**
 * Your WordDictionary object will be instantiated and called as such:
 * WordDictionary obj = new WordDictionary();
 * obj.addWord(word);
 * boolean param_2 = obj.search(word);
 */